python programming code solving